home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Ubuntu 9.10 PL / karmelkowy-koliberek-desktop-9.10-i386-PL.iso / casper / filesystem.squashfs / usr / share / hplip / setup.pyc (.txt) < prev    next >
Python Compiled Bytecode  |  2009-10-28  |  23KB  |  688 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.6)
  3.  
  4. __version__ = '9.0'
  5. __title__ = 'Printer/Fax Setup Utility'
  6. __mod__ = 'hp-setup'
  7. __doc__ = 'Installs HPLIP printers and faxes in the CUPS spooler. Tries to automatically determine the correct PPD file to use. Allows the printing of a testpage. Performs basic fax parameter setup.'
  8. import sys
  9. import getopt
  10. import time
  11. import os.path as os
  12. import re
  13. import os
  14. import gzip
  15.  
  16. try:
  17.     import readline
  18. except ImportError:
  19.     pass
  20.  
  21. from base.g import *
  22. from base import device, utils, tui, models, module
  23. from prnt import cups
  24. pm = None
  25.  
  26. def plugin_download_callback(c, s, t):
  27.     pm.update(int(100 * c * s / t), utils.format_bytes(c * s))
  28.  
  29. nickname_pat = re.compile('\\*NickName:\\s*\\"(.*)"', re.MULTILINE)
  30. USAGE = [
  31.     (__doc__, '', 'name', True),
  32.     ('Usage: %s [MODE] [OPTIONS] [SERIAL NO.|USB bus:device|IP|DEVNODE]' % __mod__, '', 'summary', True),
  33.     utils.USAGE_MODE,
  34.     utils.USAGE_GUI_MODE,
  35.     utils.USAGE_INTERACTIVE_MODE,
  36.     utils.USAGE_SPACE,
  37.     utils.USAGE_OPTIONS,
  38.     ('Automatic mode:', '-a or --auto (-i mode only)', 'option', False),
  39.     ('To specify the port on a multi-port JetDirect:', '--port=<port> (Valid values are 1\\*, 2, and 3. \\*default)', 'option', False),
  40.     ('No testpage in automatic mode:', '-x (-i mode only)', 'option', False),
  41.     ('To specify a CUPS printer queue name:', '-p<printer> or --printer=<printer> (-i mode only)', 'option', False),
  42.     ('To specify a CUPS fax queue name:', '-f<fax> or --fax=<fax> (-i mode only)', 'option', False),
  43.     ('Type of queue(s) to install:', '-t<typelist> or --type=<typelist>. <typelist>: print*, fax\\* (\\*default) (-i mode only)', 'option', False),
  44.     ('To specify the device URI to install:', '-d<device> or --device=<device> (--qt4 mode only)', 'option', False),
  45.     ('Remove printers or faxes instead of setting-up:', '-r or --rm or --remove (-u only)', 'option', False),
  46.     utils.USAGE_LANGUAGE,
  47.     utils.USAGE_LOGGING1,
  48.     utils.USAGE_LOGGING2,
  49.     utils.USAGE_LOGGING3,
  50.     utils.USAGE_HELP,
  51.     ('[SERIAL NO.|USB ID|IP|DEVNODE]', '', 'heading', False),
  52.     ('USB bus:device (usb only):', '"xxx:yyy" where \'xxx\' is the USB bus and \'yyy\' is the USB device. (Note: The \':\' and all leading zeros must be present.)', 'option', False),
  53.     ('', "Use the 'lsusb' command to obtain this information.", 'option', False),
  54.     ('IPs (network only):', 'IPv4 address "a.b.c.d" or "hostname"', 'option', False),
  55.     ('DEVNODE (parallel only):', '"/dev/parportX", X=0,1,2,...', 'option', False),
  56.     ('SERIAL NO. (usb and parallel only):', '"serial no."', 'option', True),
  57.     utils.USAGE_EXAMPLES,
  58.     ('Setup using GUI mode:', '$ hp-setup', 'example', False),
  59.     ('Setup using GUI mode, specifying usb:', '$ hp-setup -b usb', 'example', False),
  60.     ('Setup using GUI mode, specifying an IP:', '$ hp-setup 192.168.0.101', 'example', False),
  61.     ('One USB printer attached, automatic:', '$ hp-setup -i -a', 'example', False),
  62.     ('USB, IDs specified:', '$ hp-setup -i 001:002', 'example', False),
  63.     ('Network:', '$ hp-setup -i 66.35.250.209', 'example', False),
  64.     ('Network, Jetdirect port 2:', '$ hp-setup -i --port=2 66.35.250.209', 'example', False),
  65.     ('Parallel:', '$ hp-setup -i /dev/parport0', 'example', False),
  66.     ('USB or parallel, using serial number:', '$ hp-setup -i US12345678A', 'example', False),
  67.     ('USB, automatic:', '$ hp-setup -i --auto 001:002', 'example', False),
  68.     ('Parallel, automatic, no testpage:', '$ hp-setup -i -a -x /dev/parport0', 'example', False),
  69.     ('Parallel, choose device:', '$ hp-setup -i -b par', 'example', False),
  70.     utils.USAGE_SPACE,
  71.     utils.USAGE_NOTES,
  72.     ('1. If no serial number, USB ID, IP, or device node is specified, the USB and parallel busses will be probed for devices.', '', 'note', False),
  73.     ("2. Using 'lsusb' to obtain USB IDs: (example)", '', 'note', False),
  74.     ('   $ lsusb', '', 'note', False),
  75.     ('         Bus 003 Device 011: ID 03f0:c202 Hewlett-Packard', '', 'note', False),
  76.     ('   $ hp-setup --auto 003:011', '', 'note', False),
  77.     ("   (Note: You may have to run 'lsusb' from /sbin or another location. Use '$ locate lsusb' to determine this.)", '', 'note', True),
  78.     ('3. Parameters -a, -f, -p, or -t are not valid in GUI (-u) mode.', '', 'note', True),
  79.     utils.USAGE_SPACE,
  80.     utils.USAGE_SEEALSO,
  81.     ('hp-makeuri', '', 'seealso', False),
  82.     ('hp-probe', '', 'seealso', False)]
  83.  
  84. def showPasswordUI(prompt):
  85.     import getpass as getpass
  86.     print ''
  87.     print log.bold(prompt)
  88.     username = raw_input('Username: ')
  89.     password = getpass.getpass('Password: ')
  90.     return (username, password)
  91.  
  92.  
  93. def restart_cups():
  94.     if os.path.exists('/etc/init.d/cups'):
  95.         return '/etc/init.d/cups restart'
  96.     if os.path.exists('/etc/init.d/cupsys'):
  97.         return '/etc/init.d/cupsys restart'
  98.     return 'killall -HUP cupsd'
  99.  
  100. mod = module.Module(__mod__, __title__, __version__, __doc__, USAGE, (INTERACTIVE_MODE, GUI_MODE), (UI_TOOLKIT_QT3, UI_TOOLKIT_QT4), run_as_root_ok = True)
  101. (opts, device_uri, printer_name, mode, ui_toolkit, loc) = mod.parseStdOpts('axp:P:f:t:b:d:r', [
  102.     'ttl=',
  103.     'filter=',
  104.     'search=',
  105.     'find=',
  106.     'method=',
  107.     'time-out=',
  108.     'timeout=',
  109.     'printer=',
  110.     'fax=',
  111.     'type=',
  112.     'port=',
  113.     'auto',
  114.     'device=',
  115.     'rm',
  116.     'remove'], handle_device_printer = False)
  117. printer_name = None
  118. fax_name = None
  119. bus = None
  120. setup_print = True
  121. setup_fax = True
  122. makeuri = None
  123. auto = False
  124. testpage_in_auto_mode = True
  125. jd_port = 1
  126. remove = False
  127. for o, a in opts:
  128.     if o == '-x':
  129.         testpage_in_auto_mode = False
  130.         continue
  131.     if o in ('-P', '-p', '--printer'):
  132.         printer_name = a
  133.         continue
  134.     if o in ('-f', '--fax'):
  135.         fax_name = a
  136.         continue
  137.     [] if o in ('-d', '--device') else device.validateBusList(bus, False)
  138.     None if o in ('-t', '--type') else None<EXCEPTION MATCH>ValueError
  139.     if o in ('-a', '--auto'):
  140.         auto = True
  141.         continue
  142.     if o in ('-r', '--rm', '--remove'):
  143.         remove = True
  144.         continue
  145.  
  146.  
  147. try:
  148.     param = mod.args[0]
  149. except IndexError:
  150.     param = ''
  151.  
  152. log.debug('param=%s' % param)
  153. if mode == GUI_MODE:
  154.     if ui_toolkit == 'qt3':
  155.         if not utils.canEnterGUIMode():
  156.             log.error('%s requires GUI support (try running with --qt4). Also, try using interactive (-i) mode.' % __mod__)
  157.             sys.exit(1)
  158.         
  159.     elif not utils.canEnterGUIMode4():
  160.         log.error('%s requires GUI support (try running with --qt3). Also, try using interactive (-i) mode.' % __mod__)
  161.         sys.exit(1)
  162.     
  163.  
  164. if mode == GUI_MODE:
  165.     pass
  166. None if ui_toolkit == 'qt3' else None<EXCEPTION MATCH>KeyboardInterrupt
  167.  
  168. try:
  169.     cups.setPasswordCallback(showPasswordUI)
  170.     if remove:
  171.         log.error('-r/--rm/--remove not supported in -i mode.')
  172.         sys.exit(1)
  173.     
  174.     if not auto:
  175.         log.info("(Note: Defaults for each question are maked with a '*'. Press <enter> to accept the default.)")
  176.         log.info('')
  177.     
  178.     if param:
  179.         (device_uri, sane_uri, fax_uri) = device.makeURI(param, jd_port)
  180.     
  181.     if not device_uri and bus is None:
  182.         bus = tui.connection_table()
  183.         if bus is None:
  184.             sys.exit(0)
  185.         
  186.         log.info('\nUsing connection type: %s' % bus[0])
  187.         log.info('')
  188.     
  189.     if not device_uri:
  190.         device_uri = mod.getDeviceUri(device_uri, devices = device.probeDevices(bus))
  191.     
  192.     log.info(log.bold('\nSetting up device: %s\n' % device_uri))
  193.     log.info('')
  194.     print_uri = device_uri.replace('hpfax:', 'hp:')
  195.     fax_uri = device_uri.replace('hp:', 'hpfax:')
  196.     (back_end, is_hp, bus, model, serial, dev_file, host, zc, port) = device.parseDeviceURI(device_uri)
  197.     log.debug('Model=%s' % model)
  198.     mq = device.queryModelByURI(device_uri)
  199.     if not mq or mq.get('support-type', SUPPORT_TYPE_NONE) == SUPPORT_TYPE_NONE:
  200.         log.error('Unsupported printer model.')
  201.         sys.exit(1)
  202.     
  203.     if mq.get('fax-type', FAX_TYPE_NONE) in (FAX_TYPE_NONE, FAX_TYPE_NOT_SUPPORTED) and setup_fax:
  204.         setup_fax = False
  205.     
  206.     norm_model = models.normalizeModelName(model).lower()
  207.     plugin = mq.get('plugin', PLUGIN_NONE)
  208.     plugin_installed = utils.to_bool(sys_state.get('plugin', 'installed', '0'))
  209.     if plugin > PLUGIN_NONE and not plugin_installed:
  210.         tui.header('PLUG-IN INSTALLATION')
  211.         hp_plugin = utils.which('hp-plugin')
  212.         if hp_plugin:
  213.             if prop.gui_build:
  214.                 os.system('hp-plugin -i')
  215.             else:
  216.                 os.system('hp-plugin')
  217.         
  218.     
  219.     ppds = cups.getSystemPPDs()
  220.     default_model = utils.xstrip(model.replace('series', '').replace('Series', ''), '_')
  221.     stripped_model = cups.stripModel2(default_model)
  222.     installed_print_devices = device.getSupportedCUPSDevices([
  223.         'hp'])
  224.     for d in installed_print_devices.keys():
  225.         for p in installed_print_devices[d]:
  226.             log.debug("found print queue '%s'" % p)
  227.         
  228.     
  229.     installed_fax_devices = device.getSupportedCUPSDevices([
  230.         'hpfax'])
  231.     for d in installed_fax_devices.keys():
  232.         for f in installed_fax_devices[d]:
  233.             log.debug("found fax queue '%s'" % f)
  234.         
  235.     
  236.     if setup_print:
  237.         tui.header('PRINT QUEUE SETUP')
  238.         if not auto and print_uri in installed_print_devices:
  239.             log.warning('One or more print queues already exist for this device: %s.' % ', '.join(installed_print_devices[print_uri]))
  240.             (ok, setup_print) = tui.enter_yes_no('\nWould you like to install another print queue for this device', 'n')
  241.             if not ok:
  242.                 sys.exit(0)
  243.             
  244.         
  245.     
  246.     if setup_print:
  247.         if auto:
  248.             printer_name = default_model
  249.         
  250.         printer_default_model = default_model
  251.         if device_uri in installed_print_devices and printer_default_model in installed_print_devices[device_uri]:
  252.             i = 2
  253.             while True:
  254.                 t = printer_default_model + '_%d' % i
  255.                 if t not in installed_print_devices[device_uri]:
  256.                     printer_default_model += '_%d' % i
  257.                     break
  258.                 
  259.                 i += 1
  260.         
  261.         if not auto:
  262.             if printer_name is None:
  263.                 while True:
  264.                     printer_name = raw_input(log.bold("\nPlease enter a name for this print queue (m=use model name:'%s'*, q=quit) ?" % printer_default_model))
  265.                     if printer_name.lower().strip() == 'q':
  266.                         log.info('OK, done.')
  267.                         sys.exit(0)
  268.                     
  269.                     if not printer_name or printer_name.lower().strip() == 'm':
  270.                         printer_name = printer_default_model
  271.                     
  272.                     name_ok = True
  273.                     for d in installed_print_devices.keys():
  274.                         for p in installed_print_devices[d]:
  275.                             if printer_name == p:
  276.                                 log.error('A print queue with that name already exists. Please enter a different name.')
  277.                                 name_ok = False
  278.                                 break
  279.                                 continue
  280.                         
  281.                     
  282.                     for d in installed_fax_devices.keys():
  283.                         for f in installed_fax_devices[d]:
  284.                             if printer_name == f:
  285.                                 log.error('A fax queue with that name already exists. Please enter a different name.')
  286.                                 name_ok = False
  287.                                 break
  288.                                 continue
  289.                         
  290.                     
  291.                     for c in printer_name:
  292.                         if c in cups.INVALID_PRINTER_NAME_CHARS:
  293.                             log.error("Invalid character '%s' in printer name. Please enter a name that does not contain this character." % c)
  294.                             name_ok = False
  295.                             continue
  296.                     
  297.                     if name_ok:
  298.                         break
  299.                         continue
  300.             
  301.         else:
  302.             printer_name = printer_default_model
  303.         log.info('Using queue name: %s' % printer_name)
  304.         default_model = utils.xstrip(model.replace('series', '').replace('Series', ''), '_')
  305.         stripped_model = default_model.lower().replace('hp-', '').replace('hp_', '')
  306.         log.info('Locating PPD file... Please wait.')
  307.         print_ppd = cups.getPPDFile2(stripped_model, ppds)
  308.         enter_ppd = False
  309.         if print_ppd is None:
  310.             enter_ppd = True
  311.             log.error('Unable to find an appropriate PPD file.')
  312.         else:
  313.             (print_ppd, desc) = print_ppd
  314.             log.info('\nFound PPD file: %s' % print_ppd)
  315.             log.info('Description: %s' % desc)
  316.             if not auto:
  317.                 log.info('\nNote: The model number may vary slightly from the actual model number on the device.')
  318.                 (ok, ans) = tui.enter_yes_no('\nDoes this PPD file appear to be the correct one')
  319.                 if not ok:
  320.                     sys.exit(0)
  321.                 
  322.                 if not ans:
  323.                     enter_ppd = True
  324.                 
  325.             
  326.         if enter_ppd:
  327.             enter_ppd = False
  328.             (ok, enter_ppd) = tui.enter_yes_no('\nWould you like to specify the path to the correct PPD file to use', 'n')
  329.             if not ok:
  330.                 sys.exit(0)
  331.             
  332.             if enter_ppd:
  333.                 ok = False
  334.                 while True:
  335.                     user_input = raw_input(log.bold('\nPlease enter the full filesystem path to the PPD file to use (q=quit) :'))
  336.                     if user_input.lower().strip() == 'q':
  337.                         log.info('OK, done.')
  338.                         sys.exit(0)
  339.                     
  340.                     file_path = user_input
  341.                     if os.path.exists(file_path) and os.path.isfile(file_path):
  342.                         if file_path.endswith('.gz'):
  343.                             nickname = gzip.GzipFile(file_path, 'r').read(4096)
  344.                         else:
  345.                             nickname = file(file_path, 'r').read(4096)
  346.                         
  347.                         try:
  348.                             desc = nickname_pat.search(nickname).group(1)
  349.                         except AttributeError:
  350.                             desc = ''
  351.  
  352.                         if desc:
  353.                             log.info('Description for the file: %s' % desc)
  354.                         else:
  355.                             log.error("No PPD 'NickName' found. This file may not be a valid PPD file.")
  356.                         (ok, ans) = tui.enter_yes_no('\nUse this file')
  357.                         if not ok:
  358.                             sys.exit(0)
  359.                         
  360.                         if ans:
  361.                             print_ppd = file_path
  362.                         
  363.                     else:
  364.                         log.error('File not found or not an appropriate (PPD) file.')
  365.                     if ok:
  366.                         break
  367.                         continue
  368.             else:
  369.                 log.error('PPD file required. Setup cannot continue. Exiting.')
  370.                 sys.exit(1)
  371.         
  372.         if auto:
  373.             (location, info) = ('', 'Automatically setup by HPLIP')
  374.         else:
  375.             while True:
  376.                 location = raw_input(log.bold('Enter a location description for this printer (q=quit) ?'))
  377.                 if location.strip().lower() == 'q':
  378.                     log.info('OK, done.')
  379.                     sys.exit(0)
  380.                 
  381.                 break
  382.             while True:
  383.                 info = raw_input(log.bold('Enter additonal information or notes for this printer (q=quit) ?'))
  384.                 if info.strip().lower() == 'q':
  385.                     log.info('OK, done.')
  386.                     sys.exit(0)
  387.                 
  388.                 break
  389.         log.info(log.bold('\nAdding print queue to CUPS:'))
  390.         log.info('Device URI: %s' % print_uri)
  391.         log.info('Queue name: %s' % printer_name)
  392.         log.info('PPD file: %s' % print_ppd)
  393.         log.info('Location: %s' % location)
  394.         log.info('Information: %s' % info)
  395.         log.debug('Restarting CUPS...')
  396.         (status, output) = utils.run(restart_cups())
  397.         log.debug('Restart CUPS returned: exit=%d output=%s' % (status, output))
  398.         cups.setPasswordPrompt('You do not have permission to add a printer.')
  399.         if not os.path.exists(print_ppd):
  400.             (status, status_str) = cups.addPrinter(printer_name.encode('utf8'), print_uri, location, '', print_ppd, info)
  401.         else:
  402.             (status, status_str) = cups.addPrinter(printer_name.encode('utf8'), print_uri, location, print_ppd, '', info)
  403.         log.debug('addPrinter() returned (%d, %s)' % (status, status_str))
  404.         installed_print_devices = device.getSupportedCUPSDevices([
  405.             'hp'])
  406.         if print_uri not in installed_print_devices or printer_name not in installed_print_devices[print_uri]:
  407.             log.error('Printer queue setup failed. Please restart CUPS and try again.')
  408.             sys.exit(1)
  409.         
  410.     
  411.     if setup_fax and not (prop.fax_build):
  412.         log.error('Cannot setup fax - HPLIP not built with fax enabled.')
  413.         setup_fax = False
  414.     
  415.     if setup_fax:
  416.         
  417.         try:
  418.             from fax import fax
  419.         except ImportError:
  420.             setup_fax = False
  421.             log.warning('Fax setup disabled - Python 2.3+ required.')
  422.         except:
  423.             None<EXCEPTION MATCH>ImportError
  424.         
  425.  
  426.     None<EXCEPTION MATCH>ImportError
  427.     log.info('')
  428.     if setup_fax:
  429.         tui.header('FAX QUEUE SETUP')
  430.         if not auto and fax_uri in installed_fax_devices:
  431.             log.warning('One or more fax queues already exist for this device: %s.' % ', '.join(installed_fax_devices[fax_uri]))
  432.             (ok, setup_fax) = tui.enter_yes_no('\nWould you like to install another fax queue for this device', 'n')
  433.             if not ok:
  434.                 sys.exit(0)
  435.             
  436.         
  437.     
  438.     if setup_fax:
  439.         if auto:
  440.             fax_name = default_model + '_fax'
  441.         
  442.         fax_default_model = default_model + '_fax'
  443.         if fax_uri in installed_fax_devices and fax_default_model in installed_fax_devices[fax_uri]:
  444.             i = 2
  445.             while True:
  446.                 t = fax_default_model + '_%d' % i
  447.                 if t not in installed_fax_devices[fax_uri]:
  448.                     fax_default_model += '_%d' % i
  449.                     break
  450.                 
  451.                 i += 1
  452.         
  453.         if not auto:
  454.             if fax_name is None:
  455.                 while True:
  456.                     fax_name = raw_input(log.bold("\nPlease enter a name for this fax queue (m=use model name:'%s'*, q=quit) ?" % fax_default_model))
  457.                     if fax_name.lower().strip() == 'q':
  458.                         log.info('OK, done.')
  459.                         sys.exit(0)
  460.                     
  461.                     if not fax_name or fax_name.lower().strip() == 'm':
  462.                         fax_name = fax_default_model
  463.                     
  464.                     name_ok = True
  465.                     for d in installed_print_devices.keys():
  466.                         for p in installed_print_devices[d]:
  467.                             if fax_name == p:
  468.                                 log.error('A print queue with that name already exists. Please enter a different name.')
  469.                                 name_ok = False
  470.                                 break
  471.                                 continue
  472.                         
  473.                     
  474.                     for d in installed_fax_devices.keys():
  475.                         for f in installed_fax_devices[d]:
  476.                             if fax_name == f:
  477.                                 log.error('A fax queue with that name already exists. Please enter a different name.')
  478.                                 name_ok = False
  479.                                 break
  480.                                 continue
  481.                         
  482.                     
  483.                     for c in fax_name:
  484.                         if c in (' ', '#', '/', '%'):
  485.                             log.error("Invalid character '%s' in fax name. Please enter a name that does not contain this character." % c)
  486.                             name_ok = False
  487.                             continue
  488.                     
  489.                     if name_ok:
  490.                         break
  491.                         continue
  492.             
  493.         else:
  494.             fax_name = fax_default_model
  495.         log.info('Using queue name: %s' % fax_name)
  496.         fax_type = mq.get('fax-type', FAX_TYPE_NONE)
  497.         if prop.hpcups_build:
  498.             if fax_type == FAX_TYPE_SOAP:
  499.                 fax_ppd_name = 'HP-Fax2-hpcups'
  500.             else:
  501.                 fax_ppd_name = 'HP-Fax-hpcups'
  502.         elif fax_type == FAX_TYPE_SOAP:
  503.             fax_ppd_name = 'HP-Fax2-hpijs'
  504.         else:
  505.             fax_ppd_name = 'HP-Fax-hpijs'
  506.         for f in ppds:
  507.             if f.find(fax_ppd_name) >= 0:
  508.                 fax_ppd = f
  509.                 log.debug('Found PDD file: %s' % fax_ppd)
  510.                 break
  511.                 continue
  512.         else:
  513.             sys.exit(1)
  514.         if auto:
  515.             (location, info) = ('', 'Automatically setup by HPLIP')
  516.         else:
  517.             while True:
  518.                 location = raw_input(log.bold('Enter a location description for this printer (q=quit) ?'))
  519.                 if location.strip().lower() == 'q':
  520.                     log.info('OK, done.')
  521.                     sys.exit(0)
  522.                 
  523.                 break
  524.             while True:
  525.                 info = raw_input(log.bold('Enter additonal information or notes for this printer (q=quit) ?'))
  526.                 if info.strip().lower() == 'q':
  527.                     log.info('OK, done.')
  528.                     sys.exit(0)
  529.                 
  530.                 break
  531.         log.info(log.bold('\nAdding fax queue to CUPS:'))
  532.         log.info('Device URI: %s' % fax_uri)
  533.         log.info('Queue name: %s' % fax_name)
  534.         log.info('PPD file: %s' % fax_ppd)
  535.         log.info('Location: %s' % location)
  536.         log.info('Information: %s' % info)
  537.         cups.setPasswordPrompt('You do not have permission to add a fax device.')
  538.         if not os.path.exists(fax_ppd):
  539.             (status, status_str) = cups.addPrinter(fax_name.encode('utf8'), fax_uri, location, '', fax_ppd, info)
  540.         else:
  541.             (status, status_str) = cups.addPrinter(fax_name.encode('utf8'), fax_uri, location, fax_ppd, '', info)
  542.         log.debug('addPrinter() returned (%d, %s)' % (status, status_str))
  543.         installed_fax_devices = device.getSupportedCUPSDevices([
  544.             'hpfax'])
  545.         log.debug(installed_fax_devices)
  546.         if fax_uri not in installed_fax_devices or fax_name not in installed_fax_devices[fax_uri]:
  547.             log.error('Fax queue setup failed. Please restart CUPS and try again.')
  548.             sys.exit(1)
  549.         
  550.         tui.header('FAX HEADER SETUP')
  551.         if auto:
  552.             setup_fax = False
  553.         else:
  554.             while True:
  555.                 user_input = raw_input(log.bold('\nWould you like to perform fax header setup (y=yes*, n=no, q=quit) ?')).strip().lower()
  556.                 if user_input == 'q':
  557.                     log.info('OK, done.')
  558.                     sys.exit(0)
  559.                 
  560.                 if not user_input:
  561.                     user_input = 'y'
  562.                 
  563.                 setup_fax = user_input == 'y'
  564.                 if user_input in ('y', 'n', 'q'):
  565.                     break
  566.                 
  567.                 log.error("Please enter 'y' or 'n'")
  568.         if setup_fax:
  569.             d = fax.getFaxDevice(fax_uri, disable_dbus = True)
  570.             
  571.             try:
  572.                 d.open()
  573.             except Error:
  574.                 log.error('Unable to communicate with the device. Please check the device and try again.')
  575.  
  576.             
  577.             try:
  578.                 tries = 0
  579.                 ok = True
  580.                 while True:
  581.                     tries += 1
  582.                     
  583.                     try:
  584.                         current_phone_num = str(d.getPhoneNum())
  585.                         current_station_name = str(d.getStationName())
  586.                     except Error:
  587.                         log.error('Could not communicate with device. Device may be busy. Please wait for retry...')
  588.                         time.sleep(5)
  589.                         ok = False
  590.                         if tries > 12:
  591.                             break
  592.                         
  593.                         tries > 12
  594.  
  595.                     ok = True
  596.                     break
  597.                 if ok:
  598.                     while True:
  599.                         if current_phone_num:
  600.                             phone_num = raw_input(log.bold("\nEnter the fax phone number for this device (c=use current:'%s'*, q=quit) ?" % current_phone_num))
  601.                         else:
  602.                             phone_num = raw_input(log.bold('\nEnter the fax phone number for this device (q=quit) ?'))
  603.                         if phone_num.strip().lower() == 'q':
  604.                             log.info('OK, done.')
  605.                             sys.exit(0)
  606.                         
  607.                         if current_phone_num:
  608.                             if not phone_num or phone_num.strip().lower() == 'c':
  609.                                 phone_num = current_phone_num
  610.                             
  611.                         if len(phone_num) > 50:
  612.                             log.error('Phone number length is too long (>50 characters). Please enter a shorter number.')
  613.                             continue
  614.                         
  615.                         ok = True
  616.                         for x in phone_num:
  617.                             if x not in '0123456789-(+) ':
  618.                                 log.error('Invalid characters in phone number. Please only use 0-9, -, (, +, and )')
  619.                                 ok = False
  620.                                 break
  621.                                 continue
  622.                         
  623.                         if not ok:
  624.                             continue
  625.                         
  626.                         break
  627.                     while True:
  628.                         if current_station_name:
  629.                             station_name = raw_input(log.bold("\nEnter the name and/or company for this device (c=use current:'%s'*, q=quit) ?" % current_station_name))
  630.                         else:
  631.                             station_name = raw_input(log.bold('\nEnter the name and/or company for this device (q=quit) ?'))
  632.                         if station_name.strip().lower() == 'q':
  633.                             log.info('OK, done.')
  634.                             sys.exit(0)
  635.                         
  636.                         if current_station_name:
  637.                             if not station_name or station_name.strip().lower() == 'c':
  638.                                 station_name = current_station_name
  639.                             
  640.                         if len(station_name) > 50:
  641.                             log.error('Name/company length is too long (>50 characters). Please enter a shorter name/company.')
  642.                             continue
  643.                         
  644.                         break
  645.                     
  646.                     try:
  647.                         d.setStationName(station_name)
  648.                         d.setPhoneNum(phone_num)
  649.                     except Error:
  650.                         log.error('Could not communicate with device. Device may be busy.')
  651.  
  652.                     log.info('\nParameters sent to device.')
  653.             finally:
  654.                 d.close()
  655.  
  656.         
  657.     
  658.     if setup_print:
  659.         print_test_page = False
  660.         tui.header('PRINTER TEST PAGE')
  661.         if auto:
  662.             if testpage_in_auto_mode:
  663.                 print_test_page = True
  664.             
  665.         else:
  666.             (ok, print_test_page) = tui.enter_yes_no('\nWould you like to print a test page')
  667.             if not ok:
  668.                 sys.exit(0)
  669.             
  670.         if print_test_page:
  671.             path = utils.which('hp-testpage')
  672.             if printer_name:
  673.                 param = '-p%s' % printer_name
  674.             else:
  675.                 param = '-d%s' % print_uri
  676.             if len(path) > 0:
  677.                 cmd = 'hp-testpage %s' % param
  678.             else:
  679.                 cmd = 'python ./testpage.py %s' % param
  680.             log.debug(cmd)
  681.             os.system(cmd)
  682.         
  683. except KeyboardInterrupt:
  684.     log.error('User exit')
  685.  
  686. log.info('')
  687. log.info('Done.')
  688.